class Agent_Memory:
    def __init__(self, user_identity: str = ''):
        self.user_identity: str = user_identity
        self.query_record: List[str] = []

    def add_query(self, query: str):
        self.query_record.append(query)

    def merge_memory(self):
        if not self.query_record:
            return "用户还没有提问过问题。"
        history_content = "用户已经进行了以下提问：\n"
        for i, query in enumerate(self.query_record):
            history_content += "第{}次提问：{}\n".format(i+1, query)
        
        #print(f"history_content: {history_content}")
        return history_content


# ============================================================================
# AI Interaction
# ============================================================================
# 角色设定
agent_role_design = '''你是一个熟悉公司经营管理流程的总裁助理，你和你服务的高管已经共事超过5年；
你非常了解经营者们的工作习惯，当他们向你提出了一些关于公司经营状况的问题后，你不仅能快速用关键数据信息回复并且能够猜到他们后续还会有哪些疑问；
现在你要运用你的能力来完成一些特定的工作。
'''

remake_query_prompt_template = '''一位{}的用户还没有想好下面要向你提出什么问题，你需要给他推荐一些他可能感兴趣的提问。
{}

你推荐的问题不能和用户已经提出的问题出现重复，这种重复指的是语义上的重复而不是仅指内容上的重复；你推荐的问题要确保和之前用户提出的问题在同一个经营领域，不能太过跳脱，导致推荐了用户根本不感兴趣的问题。
请按照下面的json格式输出你推荐的提问，记住不要包含任何json以外的内容：
[
  {{
    "qid": "1",
    "content": ""
  }},
  {{
    "qid": "2",
    "content": ""
  }},
  {{
    "qid": "3",
    "content": ""
  }}
]

推荐至少三个提问，但不要超过八个。
'''